#!/usr/bin/env python3
# -*- coding:utf-8 -*-
# @Author: renjin@bit.edu.cn
# @Date  : 2024-08-06

from spirecv.dataloader.COCODatasetLoaderNode import COCODatasetLoaderNode
from spirecv.algorithm.common_det.YOLOv10DetNode import YOLOv10DetNode
from spirecv.evaluator.COCOResultAssembleNode import COCOResultAssembleNode
from spirecv.evaluator.COCOEvalNode import COCOEvalNode


job_name = 'detection_evaluation'
parameter_file = '../../params/spirecv2/default_params.json'


dataloader = COCODatasetLoaderNode(job_name, param_dict_or_file=parameter_file)
algorithm = YOLOv10DetNode(job_name, param_dict_or_file=parameter_file)
assembler = COCOResultAssembleNode(job_name, param_dict_or_file=parameter_file)
evaluator = COCOEvalNode(job_name, param_dict_or_file=parameter_file)


dataloader.launch_next()
